# from convert_model import ConvertModel
# from flip_convert import FlipConvert
# from rotation_convert import RotationConvert
# from offset_convert import OffsetConvert
# from local_occlusion_convert import LocalOcclusionConvert
# from padding_resize_convert import PaddingResizeConvert
# from scale_convert import ScaleConvert
# from splicing_convert import SplicingConvert
import cv2 as cv
# from padding_resize_convert import PaddingResizeConvert
# from mixup_convert import MixUpConvert
# import numpy as np


# converts = ConvertModel(
#     # LocalOcclusionConvert(use_rate=1, min_size=20, max_size=50, fill_color=[114, 114, 114]),
#     # PaddingResizeConvert((512, 512), use_rate=1, padding_value=114),
#     # FlipConvert(use_rate=1),
#     # RotationConvert(use_rate=1),
#     # OffsetConvert(use_rate=1, min_offset=50, max_offset=200),
#     ScaleConvert(use_rate=1)
# )

img = cv.imread(r'./test_imgs/1.jpg')
cv.imshow('org', img)

# img, _, _ = converts.convert(img, None, None)

# cv.imshow('convert', img)
# cv.waitKey(0)



# sp = SplicingConvert(img_count=4, use_rate=1)
# img1 = cv.imread(r'./test_imgs/1.jpg')
# img2 = cv.imread(r'./test_imgs/2.jpg')
# img3 = cv.imread(r'./test_imgs/3.jpg')
# img4 = cv.imread(r'./test_imgs/4.jpg')

# resize = PaddingResizeConvert((512, 512))
# img1, _, _ = resize.convert(img1, None, None)
# img2, _, _ = resize.convert(img2, None, None)
# img3, _, _ = resize.convert(img3, None, None)
# img4, _, _ = resize.convert(img4, None, None)

# res, _, _, _ = sp.convert([img1, img2, img3, img4], None, None)
# mix = MixUpConvert()
# img12, _, _ = mix.convert([img1, img2], None, None)
# img34, _, _ = mix.convert([img3, img4], None, None)


# cv.imshow('convert', res)
# cv.waitKey(0)

# cv.imshow('img12', img12)
# cv.waitKey(0)

# cv.imshow('img34', img34)
# cv.waitKey(0)


from hsv_convert import HSVConvert
hsv = HSVConvert(1, True, 0)

img, _, _ = hsv.convert(img, None, None)

cv.imshow('img1', img)
cv.waitKey(0)



